from langchain.prompts import PromptTemplate
from langchain_openai import OpenAI
from langchain.chains import LLMChain
 
llm = OpenAI(temperature=0.9,
             base_url="https://api.gpts.vin/v1",
             api_key="sk-ELq6vE96xLlCHWjR21D8D93168604d2f975dF2572683464f")
prompt = PromptTemplate(
    input_variables=["product"],
    template="What is a good name for a company that makes {product}?",
)

chain = LLMChain(llm=llm, prompt=prompt)
print(chain.invoke("colorful socks"))
